import torch
import torch.fx
import torchvision.models as models
import statistics,tabulate,time

rn18 = models.resnet18()
rn18.eval()

traced_rn18 = torch.fx.symbolic_trace(rn18)
print(traced_rn18.graph)